package com.linyaonan.leetcode.easy._1385;

import java.util.Arrays;

/**
 * 给你两个整数数组 arr1 ， arr2 和一个整数 d ，请你返回两个数组之间的 距离值 。
 *
 * 「距离值」 定义为符合此距离要求的元素数目：对于元素 arr1[i] ，不存在任何元素 arr2[j] 满足 |arr1[i]-arr2[j]| <= d 。
 *
 *  
 *
 * 示例 1：
 *
 * 输入：arr1 = [4,5,8], arr2 = [10,9,1,8], d = 2
 * 输出：2
 * 解释：
 * 对于 arr1[0]=4 我们有：
 * |4-10|=6 > d=2
 * |4-9|=5 > d=2
 * |4-1|=3 > d=2
 * |4-8|=4 > d=2
 * 所以 arr1[0]=4 符合距离要求
 *
 * 对于 arr1[1]=5 我们有：
 * |5-10|=5 > d=2
 * |5-9|=4 > d=2
 * |5-1|=4 > d=2
 * |5-8|=3 > d=2
 * 所以 arr1[1]=5 也符合距离要求
 *
 * 对于 arr1[2]=8 我们有：
 * |8-10|=2 <= d=2
 * |8-9|=1 <= d=2
 * |8-1|=7 > d=2
 * |8-8|=0 <= d=2
 * 存在距离小于等于 2 的情况，不符合距离要求
 *
 * 故而只有 arr1[0]=4 和 arr1[1]=5 两个符合距离要求，距离值为 2
 * 示例 2：
 *
 * 输入：arr1 = [1,4,2,3], arr2 = [-4,-3,6,10,20,30], d = 3
 * 输出：2
 * 示例 3：
 *
 * 输入：arr1 = [2,1,100,3], arr2 = [-5,-2,10,-3,7], d = 6
 * 输出：1
 *  
 *
 * 提示：
 *
 * 1 <= arr1.length, arr2.length <= 500
 * -10^3 <= arr1[i], arr2[j] <= 10^3
 * 0 <= d <= 100
 *
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode.cn/problems/find-the-distance-value-between-two-arrays
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 */
class FindTheDistanceValueBetweenTwoArrays {

    /**
     * 暴力求解即可，二分法也得先进行排序，500样本的情况下速率差不多
     * @param arr1
     * @param arr2
     * @param d
     * @return
     */
    public int findTheDistanceValue(int[] arr1, int[] arr2, int d) {
        //初始化时假设arr1中的所有元素都满足题意
        int res=arr1.length;
        for (int a1 : arr1) {
            for (int a2 : arr2) {
                //假如 |a1 - a2|<=d ,说明a1不合题意 整体结果数减一
                if (Math.abs(a1 - a2) <= d) {
                    res--;
                    break;
                }
            }
        }
        return res;
    }

    public int findTheDistanceValue2(int[] arr1, int[] arr2, int d) {
        // arr1 = [1,4,2,3], arr2 = [-4,-3,6,10,20,30], d = 3

        // arr2 先排序：[-4,-3,6,10,20,30]

        Arrays.sort(arr2);
        int res = arr1.length;
        for (int i : arr1) {
            if (check(arr2, i, d)!= -1) {
                res--;
            }
        }

        return res;
    }

    /**
     * 数组arr中查询索引，满足 |arr[i] - t| <= d
     *
     * @param arr
     * @param t
     * @param d
     * @return
     */
    public int check(int[] arr, int t, int d) {
        int l = 0;
        int r = arr.length - 1;

        while (l <= r) {
            int m = (l + r) >>> 1;
            int temp = arr[m];
            // 分为两种情况，tmp比t大，tmp比t小
            if (temp > t) {
                if (temp - t <= d) {
                    return m;
                }
                r = m - 1;
            } else if (temp < t) {
                if (t - temp <= d) {
                    return m;
                }
                l = m + 1;
            } else {
                // 如果temp == t 则表示找到索引
                return m;
            }
        }
        return -1;
    }
}